package gsdb

import (
	"fmt"

	"gorm.io/gorm"
)

// 注册 mysql 连接。
//
//	@configName  mysql 配置根节点。
func Register(config map[string]*MysqlConfig) {
	MysqlManager = newMysqlManager()
	MysqlManager.init(config)
}

// 获取 db 实例。
//
//	@nodeName db 节点名称。
func Get(nodeName string) *gorm.DB {
	if MysqlManager == nil {
		panic("mysql service is not register!")
	}

	if v, ok := MysqlManager.dbMap[nodeName]; ok {
		return v
	}

	panic(fmt.Sprintf("db %s does not exist", nodeName))
}
